how to generate an array of random numbers in python